package com.wcj.linktable.classicquestion;

import com.wcj.linktable.ListNode;

/**
 * 给你一个链表的头节点 head 和一个整数 val ，
 * 请你删除链表中所有满足 Node.val == val 的节点，并返回 新的头节点 。
 *
 * 输入：head = [1,2,6,3,4,5,6], val = 6
 * 输出：[1,2,3,4,5]
 */
public class 移除链表元素 {

    public ListNode removeElements(ListNode head, int val) {
        if (head == null){
            return head;
        }
        ListNode pre = null;
        ListNode cur = head;
        while (cur != null){
            if (cur.val == val){
                if (pre != null){
                    pre.next = cur.next;
                    cur = cur.next;
                }else {
                    head = head.next;
                    cur = head;
                }
            }else {
                pre = cur;
                cur = cur.next;
            }
        }
        return head;
    }
}
